"""
演示函数综合案例开发
"""
# 定义全局变量 money name
money = 5000000
name = None
# 要求客户输入姓名
name = input("请输入您的姓名：")
# 定义查询函数
def query(show_header):
    if show_header:
        print("------------------查询余额------------------")
    print(f"{name},你好，你的余额为：{money}")

# 定义存款函数
def saving(save_money):
    print("------------------存款------------------")
    global money
    money += save_money
    print(f"{name}, 你好，你存款{save_money}元成功")
    # 调用query函数查询余额
    query(False)

# 定义取款函数
def withdraw(withdraw_money):
    print("------------------取款------------------")
    global  money
    money -= withdraw_money
    print(f"{name}, 你好，你取款{money}元成功。")
    # 调用query函数查询余额
    query(False)

# 定义主菜单函数
def main():
    print("------------------主菜单------------------")
    print(f"{name}, 你好，欢迎来到黑马银行ATM。请选择操作：")
    print("查询余额\t[输入1]")
    print("存款\t\t[输入2]")
    print("取款\t\t[输入3]")
    print("退出\t\t[输入4]")
    return input("请输入你的选择")

while True:
    keyboard_input = main()
    if keyboard_input == "1":
        query(True)
        continue # 通过continue继续下一次循环，一进来就是回到了主菜单
    elif keyboard_input == "2":
        save_money = int(input("请输入你的存款金额: "))
        saving(save_money)
        continue
    elif keyboard_input == "3":
        withdraw_money = int(input("请输入你的取款金额: "))
        withdraw(withdraw_money)
        continue
    else:
        print("退出程序")
        break # 通过break退出循环
